package org.jivesoftware.smack.proxy;

import javax.net.SocketFactory;

/**
 * Class which stores proxy information such as proxy type, host, port, 
 * authentication etc.
 * 
 * @author Atul Aggarwal
 */
public class ProxyInfo {

    public /*enum*/ static class ProxyType {

        public static final ProxyType NONE = new ProxyType();
        public static final ProxyType HTTP = new ProxyType();
        public static final ProxyType SOCKS4 = new ProxyType();
        public static final ProxyType SOCKS5 = new ProxyType();
    }
    private String proxyAddress;
    private int proxyPort;
    private String proxyUsername;
    private String proxyPassword;
    private ProxyType proxyType;

    public ProxyInfo(ProxyType pType, String pHost, int pPort, String pUser,
            String pPass) {
        this.proxyType = pType;
        this.proxyAddress = pHost;
        this.proxyPort = pPort;
        this.proxyUsername = pUser;
        this.proxyPassword = pPass;
    }

    public static ProxyInfo forHttpProxy(String pHost, int pPort, String pUser,
            String pPass) {
        return new ProxyInfo(ProxyType.HTTP, pHost, pPort, pUser, pPass);
    }

    public static ProxyInfo forSocks4Proxy(String pHost, int pPort, String pUser,
            String pPass) {
        return new ProxyInfo(ProxyType.SOCKS4, pHost, pPort, pUser, pPass);
    }

    public static ProxyInfo forSocks5Proxy(String pHost, int pPort, String pUser,
            String pPass) {
        return new ProxyInfo(ProxyType.SOCKS5, pHost, pPort, pUser, pPass);
    }

    public static ProxyInfo forNoProxy() {
        return new ProxyInfo(ProxyType.NONE, null, 0, null, null);
    }

    public static ProxyInfo forDefaultProxy() {
        return new ProxyInfo(ProxyType.NONE, null, 0, null, null);
    }

    public ProxyType getProxyType() {
        return proxyType;
    }

    public String getProxyAddress() {
        return proxyAddress;
    }

    public int getProxyPort() {
        return proxyPort;
    }

    public String getProxyUsername() {
        return proxyUsername;
    }

    public String getProxyPassword() {
        return proxyPassword;
    }

    public SocketFactory getSocketFactory() {
        if (proxyType == ProxyType.NONE) {
            return new DirectSocketFactory();
        } else if (proxyType == ProxyType.HTTP) {
            return new HTTPProxySocketFactory(this);
        } else if (proxyType == ProxyType.SOCKS4) {
            return new Socks4ProxySocketFactory(this);
        } else if (proxyType == ProxyType.SOCKS5) {
            return new Socks5ProxySocketFactory(this);
        } else {
            return null;
        }
    }
}
